# _*_ coding: utf-8 _*_
"""
@ 时间    ：2024/10/25 12:08
@ 作者    ：旺财
@ 文件    ：非数值类型处理.py
@ 说明    ：   
"""
import pandas as pd
from sklearn.preprocessing import LabelEncoder

# 1. map方式
df = pd.DataFrame({'客户编号': [1, 2, 3], '性别': ['男', '女', '男']})
print(df)
df['性别'] = df['性别'].map(lambda x: 1 if x == '男' else 0)
print(df)

# 2. Label Encoding方式
df = pd.DataFrame({'客户编号': [1, 2, 3, 4, 5], '城市': ['北京', '上海', '广州', '深圳', '北京']})
print(df)
df['城市'] = LabelEncoder().fit_transform(df['城市'])
print(df)